package 中等.树;

import util.TreeNode;

/**
 * 给定一个不重复的整数数组 nums 。 最大二叉树 可以用下面的算法从 nums 递归地构建:
 * 创建一个根节点，其值为 nums 中的最大值。
 * 递归地在最大值 左边 的 子数组前缀上 构建左子树。
 * 递归地在最大值 右边 的 子数组后缀上 构建右子树。
 * 返回 nums 构建的 最大二叉树 。
 * <p>
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode.cn/problems/maximum-binary-tree
 */
public class 最大二叉树_654 {

    public static void main(String[] args) {

        最大二叉树_654 maxTree = new 最大二叉树_654();
        int[] nums = {3, 2, 1, 6, 0, 5};
        maxTree.constructMaximumBinaryTree(nums);

    }

    public TreeNode constructMaximumBinaryTree(int[] nums) {
        TreeNode root = new TreeNode();
        int maxIdx = 0;
        for (int i = 1; i < nums.length; i++) {
            if (nums[i] > nums[maxIdx]) {
                maxIdx = i;
            }
        }
        root.val = nums[maxIdx];
        buildTree(nums, root, 0, nums.length - 1, maxIdx);
        return root;
    }

    /**
     * 表示从数组 [left,right] 中的最大值索引为 maxIdx
     */
    public void buildTree(int[] nums, TreeNode root, int left, int right, int maxIdx) {
        // 停止条件
        if (left >= right) return;
        // 构建从 [left,maxIdx) 的数组范围的树
        if (maxIdx > left) {
            int lMaxIdx = left;
            for (int i = left; i < maxIdx; i++) {
                if (nums[i] > nums[lMaxIdx]) {
                    lMaxIdx = i;
                }
            }
            TreeNode leftNode = new TreeNode(nums[lMaxIdx]);
            root.left = leftNode;
            buildTree(nums, leftNode, left, maxIdx - 1, lMaxIdx);
        }
        // 构建从 (maxIdx,right] 的数组范围的树
        if (maxIdx < right) {
            int rMaxIdx = maxIdx + 1;
            for (int i = maxIdx + 1; i <= right; i++) {
                if (nums[i] > nums[rMaxIdx]) {
                    rMaxIdx = i;
                }
            }
            TreeNode rightNode = new TreeNode(nums[rMaxIdx]);
            root.right = rightNode;
            buildTree(nums, rightNode, maxIdx + 1, right, rMaxIdx);
        }
    }
}
